How can I fix 'TypeError: cannot unpack non-iterable NoneType' in my Python state machine